package org.jsola.contract.helper;

import org.jsola.common.SafeKit;
import org.jsola.contract.dto.ContractVariableAddDTO;
import org.jsola.contract.dto.ContractVariableUpdateDTO;
import org.jsola.contract.entity.ContractVariableDO;
import org.jsola.contract.query.ContractVariableQuery;
import org.jsola.contract.service.IContractVariableService;
import org.jsola.contract.vo.ContractVariableVO;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;

import java.util.List;

import static org.jsola.contract.core.AbstractTest.SITE_ID;



/**
 * 合同关联模板变量
 *
 * @author JanusMix
 */
@Component
public class ContractVariableTestHelper extends BaseTestHelper {
    
    @Autowired
    private IContractVariableService contractVariableService;
    @Autowired
    private VariableTestHelper variableTestHelper;

    /**
     * 生成add测试数据
     * @return ContractVariableAddDTO
     */
    public ContractVariableAddDTO getContractVariableAddDTO(){
        ContractVariableAddDTO contractVariableAddDTO = new ContractVariableAddDTO();
        contractVariableAddDTO.setContractId(SafeKit.getLong(getRandomValue("bigint",11)));
        contractVariableAddDTO.setVariableId(variableTestHelper.saveVariableExample().getId());
        contractVariableAddDTO.setContent(getRandomValue("varchar",1000));
        return contractVariableAddDTO;
    }


    /**
     * 生成update测试数据
     * @return ContractVariableUpdateDTO
     */
    public ContractVariableUpdateDTO getContractVariableUpdateDTO(ContractVariableVO contractVariableVO){
        return contractVariableService.selectDOById(contractVariableVO.getId(), SITE_ID).to(ContractVariableUpdateDTO.class);
    }

    /**
     * 生成查询数据
     * @return ContractVariableQuery
     */
    public ContractVariableQuery getContractVariableQuery() {
        ContractVariableQuery contractVariableQuery = new ContractVariableQuery();
        List<ContractVariableDO> contractVariableDOList = contractVariableService.list(new ContractVariableDO(),SITE_ID);
        if (!CollectionUtils.isEmpty(contractVariableDOList)) {
            ContractVariableDO contractVariableDO = contractVariableDOList.get(0);
            contractVariableQuery = contractVariableDO.to(ContractVariableQuery.class);
        }
        return contractVariableQuery;
    }

    /**
     * 保存示例
     * @return ContractVariableVO
     */
    public ContractVariableVO saveContractVariableExample(){
        ContractVariableAddDTO addDTO = this.getContractVariableAddDTO();
        ContractVariableVO contractVariableVO = contractVariableService.save(addDTO,buildTokenUser());
        return contractVariableVO;
    }


}

